package com.moon.leetcode;

// 714. 买卖股票的最佳时机含手续费
// 
//给定一个整数数组 prices，其中第 i 个元素代表了第 i 天的股票价格 ；非负整数 fee 代表了交易股票的手续费用。 
//
// 你可以无限次地完成交易，但是你每笔交易都需要付手续费。如果你已经购买了一个股票，在卖出它之前你就不能再继续购买股票了。 
//
// 返回获得利润的最大值。 
//
// 注意：这里的一笔交易指买入持有并卖出股票的整个过程，每笔交易你只需要为支付一次手续费。 
//
// 示例 1: 
//
// 输入: prices = [1, 3, 2, 8, 4, 9], fee = 2
//输出: 8
//解释: 能够达到的最大利润:  
//在此处买入prices[0] = 1
//在此处卖出 prices[3] = 8
//在此处买入 prices[4] = 4
//在此处卖出 prices[5] = 9
//总利润:((8 - 1) - 2) + ((9 - 4) - 2) = 8. 
//
// 注意: 
//
// 0 < prices.length <= 50000.
// 0 < prices[i] < 50000. 
// 0 <= fee < 50000. 
// 
// Related Topics 贪心算法 数组 动态规划 
// 👍 452 👎 0
public class No714_maxProfit {

    public static void main(String[] args) {
        System.out.println(new No714_maxProfit().maxProfit_v2(new int[]{2, 1, 4, 4, 2, 3, 2, 5, 1, 2}, 1));
    }

    public int maxProfit_v1(int[] prices, int fee) {
        int buy = -prices[0], sell = Math.max(0, buy + prices[0] - fee);
        for (int price : prices) {
            buy = Math.max(buy, sell - price);
            sell = Math.max(sell, buy + price - fee);
        }
        return sell;
    }

    public int maxProfit_v2(int[] prices, int fee) {
        Integer[][] mem = new Integer[prices.length][2];
        return traverseByMem(prices, false, 0, fee, mem);
    }

    private int traverseByMem(int[] prices, boolean s, int day, int fee, Integer[][] mem) {
        if (day == prices.length) {
            return 0;
        }
        if (!s && mem[day][0] != null) {
            return mem[day][0];
        }
        if (s && mem[day][1] != null) {
            return mem[day][1];
        }
        // 什么都不做
        int pass = traverseByMem(prices, s, day + 1, fee, mem);
        // 买入
        int buy = traverseByMem(prices, true, day + 1, fee, mem) - prices[day];
        // 卖出
        int sell = traverseByMem(prices, false, day + 1, fee, mem) + prices[day] - fee;
        return s ? (mem[day][1] = Math.max(pass, sell)) : (mem[day][0] = Math.max(pass, buy));
    }
}
